#pragma once

#include <iostream>
#include <vector>
#include <string>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

using std::string;
namespace ns_util
{
    class TimeUtil
    {
    public:
        // 获取秒时间戳
        static string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr); // 第二个参数已经废弃，传nullptr即可
            return std::to_string(_time.tv_sec);
        }
        // 获取毫秒级时间戳
        static string GetTimeMS()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000); // 转为ms (s*1000+us/1000)
        }
    };

    const string temp_path = "./temp/";
    // 123->./temp/123.cpp
    // 123->./temp/123.exe
    // 123->./temp/123.stderr
    class PathUtil
    {
    public:
        // 添加路径及后缀
        static string AddSuffix(const string &file_name, const string &suffix)
        {
            string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        // 下面三个是编译时需要的临时文件
        // 构建源文件路径+后缀的完整文件名
        static string Src(const string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 构建可执行程序的完整路径+后缀名
        static string Exe(const string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        static string CompileError(const string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 运行时需要的临时文件
        static string Stdin(const string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static string Stdout(const string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        // 构建该程序对应的标准错误完整的路径+后缀名
        static string Stderr(const string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const string &path_name)
        {
            struct stat st;
            // 0 成功 ，-1失败
            if (stat(path_name.c_str(), &st) == 0)
            {
                // 获取属性成功，文件存在
                return true;
            }
            return false;
        }
        static string UniqFileName()
        {
            // 通过毫秒级时间戳+原子性递增唯一值, 来保证唯一性
            static std::atomic_uint id(0);
            id++;
            string ms = TimeUtil::GetTimeMS();
            string uniq_id = std::to_string(id);
            return ms + "_" + uniq_id;
        }
        static bool WriteFile(const string &target, const string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            out << content; // 把content写入流中
            out.close();
            return true;
        }
        static bool ReadFile(const string &target, string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            string line;
            // getline不保存行分隔符，有时候需要保留\n，所以用keep变量控制是否保留
            // getline内部重载了强制类型转化
            while (getline(in, line))
            {
                // 拼接
                (*content) += line;
                (*content) += keep ? "\n" : "";
            }
            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        // 分割行字符串
        static void SplitString(const string &str, std::vector<string> *target, const string &sep)
        {
            // str: 输入型，目标要切分的字符串
            // target: 输出型，保存切分完毕的结果
            // sep: 指定的分割符集合
            // token_compress_on压缩连续分隔符 eg. 分隔符':' ,hello:::::world
            boost::split(*target, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
};